package depthorder;

import entity.TreeNode;

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Deque;
import java.util.List;

public class PreOrderTraversal {
    /*
    * 二叉树的前序遍历
    * 给你二叉树的根节点 root ，返回它节点值的 前序 遍历。
    *
    * 树中节点数目在范围 [0, 100] 内
    * -100 <= Node.val <= 100
    * */
    public static void main(String[] args){

    }

    // 我的想法：递归
    public List<Integer> mySolution(TreeNode root){
        List<Integer> list = new ArrayList<>();
        preOrderTraversal(root, list);
        return list;
    }

    public void preOrderTraversal(TreeNode cur, List<Integer> list){
        if(cur == null)
            return;

        list.add(cur.val);  // 先中
        preOrderTraversal(cur.left, list); // 再左
        preOrderTraversal(cur.right, list); // 再右
    }

    // 迭代, 根节点入栈。然后开始循环。 循环体中，先弹栈，获取节点值，再将节点的左右儿子压栈
    // 注意对于先遍历左儿子，再遍历右儿子，栈后进先出，所以先压右儿子，再压左儿子。
    public List<Integer> MySolution1(TreeNode root){
        List<Integer> list = new ArrayList<>();
        Deque<TreeNode> stack = new ArrayDeque<>();
        if(root == null)
            return list;
        stack.add(root);
        TreeNode cur;
        while (!stack.isEmpty()){
            cur = stack.pollLast();
            list.add(cur.val);
            if(cur.right != null)
                stack.add(cur.right);
            if(cur.left != null)
                stack.add(cur.left);
        }
        return list;
    }
}

